

<!DOCTYPE html>
<html class="writer-html5" lang="en" >
<head>
  <meta charset="utf-8" />
  
  <meta name="viewport" content="width=device-width, initial-scale=1.0" />
  
  <title>Index &mdash; WarpDrive 1.0 documentation</title>
  

  
  <link rel="stylesheet" href="_static/css/theme.css" type="text/css" />
  <link rel="stylesheet" href="_static/pygments.css" type="text/css" />
  <link rel="stylesheet" href="_static/pygments.css" type="text/css" />
  <link rel="stylesheet" href="_static/css/theme.css" type="text/css" />
  <link rel="stylesheet" href="https://assets.readthedocs.org/static/css/badge_only.css" type="text/css" />

  
  

  
  

  

  
  <!--[if lt IE 9]>
    <script src="_static/js/html5shiv.min.js"></script>
  <![endif]-->
  
    
      <script type="text/javascript" id="documentation_options" data-url_root="./" src="_static/documentation_options.js"></script>
        <script data-url_root="./" id="documentation_options" src="_static/documentation_options.js"></script>
        <script src="_static/jquery.js"></script>
        <script src="_static/underscore.js"></script>
        <script src="_static/doctools.js"></script>
        <script async="async" src="https://assets.readthedocs.org/static/javascript/readthedocs-doc-embed.js"></script>
    
    <script type="text/javascript" src="_static/js/theme.js"></script>

    
    <link rel="index" title="Index" href="#" />
    <link rel="search" title="Search" href="search.html" /> 

<!-- RTD Extra Head -->

<link rel="stylesheet" href="https://assets.readthedocs.org/static/css/readthedocs-doc-embed.css" type="text/css" />

<script type="application/json" id="READTHEDOCS_DATA">{"ad_free": "", "api_host": "", "build_date": "2021-08-31T22:53:14Z", "builder": "sphinx", "canonical_url": "", "commit": "", "docroot": "", "features": {"docsearch_disabled": false}, "global_analytics_code": null, "language": "", "page": "genindex", "programming_language": "", "project": "", "source_suffix": "", "subprojects": {}, "theme": "", "user_analytics_code": null, "version": ""}</script>

<!--
Using this variable directly instead of using `JSON.parse` is deprecated.
The READTHEDOCS_DATA global variable will be removed in the future.
-->
<script type="text/javascript">
READTHEDOCS_DATA = JSON.parse(document.getElementById('READTHEDOCS_DATA').innerHTML);
</script>

<script type="text/javascript" src="https://assets.readthedocs.org/static/javascript/readthedocs-analytics.js" async="async"></script>

<!-- end RTD <extrahead> -->
</head>

<body class="wy-body-for-nav">

   
  <div class="wy-grid-for-nav">
    
    <nav data-toggle="wy-nav-shift" class="wy-nav-side">
      <div class="wy-side-scroll">
        <div class="wy-side-nav-search" >
          

          
            <a href="index.html" class="icon icon-home"> WarpDrive
          

          
          </a>

          
            
            
          

          
<div role="search">
  <form id="rtd-search-form" class="wy-form" action="search.html" method="get">
    <input type="text" name="q" placeholder="Search docs" />
    <input type="hidden" name="check_keywords" value="yes" />
    <input type="hidden" name="area" value="default" />
  </form>
</div>

          
        </div>

        
        <div class="wy-menu wy-menu-vertical" data-spy="affix" role="navigation" aria-label="main navigation">
          
            
            
              
            
            
              <p class="caption" role="heading"><span class="caption-text">Contents:</span></p>
<ul>
<li class="toctree-l1"><a class="reference internal" href="warp_drive.html">warp_drive package</a></li>
<li class="toctree-l1"><a class="reference internal" href="example_envs.html">example_envs package</a></li>
</ul>

            
          
        </div>
        
      </div>
    </nav>

    <section data-toggle="wy-nav-shift" class="wy-nav-content-wrap">

      
      <nav class="wy-nav-top" aria-label="top navigation">
        
          <i data-toggle="wy-nav-top" class="fa fa-bars"></i>
          <a href="index.html">WarpDrive</a>
        
      </nav>


      <div class="wy-nav-content">
        
        <div class="rst-content">
        
          

















<div role="navigation" aria-label="breadcrumbs navigation">

  <ul class="wy-breadcrumbs">
    
      <li><a href="index.html" class="icon icon-home"></a> &raquo;</li>
        
      <li>Index</li>
    
    
      <li class="wy-breadcrumbs-aside">
        
          
        
      </li>
    
  </ul>

  
  <hr/>
</div>
          <div role="main" class="document" itemscope="itemscope" itemtype="http://schema.org/Article">
           <div itemprop="articleBody">
            

<h1 id="index">Index</h1>

<div class="genindex-jumpbox">
 <a href="#A"><strong>A</strong></a>
 | <a href="#B"><strong>B</strong></a>
 | <a href="#C"><strong>C</strong></a>
 | <a href="#D"><strong>D</strong></a>
 | <a href="#E"><strong>E</strong></a>
 | <a href="#F"><strong>F</strong></a>
 | <a href="#G"><strong>G</strong></a>
 | <a href="#H"><strong>H</strong></a>
 | <a href="#I"><strong>I</strong></a>
 | <a href="#K"><strong>K</strong></a>
 | <a href="#L"><strong>L</strong></a>
 | <a href="#M"><strong>M</strong></a>
 | <a href="#N"><strong>N</strong></a>
 | <a href="#O"><strong>O</strong></a>
 | <a href="#P"><strong>P</strong></a>
 | <a href="#R"><strong>R</strong></a>
 | <a href="#S"><strong>S</strong></a>
 | <a href="#T"><strong>T</strong></a>
 | <a href="#U"><strong>U</strong></a>
 | <a href="#W"><strong>W</strong></a>
 
</div>
<h2 id="A">A</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="warp_drive.training.algorithms.html#warp_drive.training.algorithms.a2c.A2C">A2C (class in warp_drive.training.algorithms.a2c)</a>
</li>
      <li><a href="warp_drive.utils.html#warp_drive.utils.constants.Constants.ACTIONS">ACTIONS (warp_drive.utils.constants.Constants attribute)</a>
</li>
      <li><a href="warp_drive.utils.html#warp_drive.utils.data_feed.DataFeed.add_data">add_data() (warp_drive.utils.data_feed.DataFeed method)</a>
</li>
      <li><a href="data_manager.html#data_manager.CUDADataManager.add_meta_info">add_meta_info() (data_manager.CUDADataManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CUDADataManager.add_meta_info">(warp_drive.managers.data_manager.CUDADataManager method)</a>
</li>
      </ul></li>
  </ul></td>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="data_manager.html#data_manager.CUDADataManager.add_shared_constants">add_shared_constants() (data_manager.CUDADataManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CUDADataManager.add_shared_constants">(warp_drive.managers.data_manager.CUDADataManager method)</a>
</li>
      </ul></li>
      <li><a href="warp_drive.training.utils.html#warp_drive.training.utils.data_loader.all_equal">all_equal() (in module warp_drive.training.utils.data_loader)</a>
</li>
      <li><a href="function_manager.html#function_manager.CUDASampler.assign">assign() (function_manager.CUDASampler method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDASampler.assign">(warp_drive.managers.function_manager.CUDASampler method)</a>
</li>
      </ul></li>
  </ul></td>
</tr></table>

<h2 id="B">B</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="function_manager.html#function_manager.CUDAFunctionManager.block">block (function_manager.CUDAFunctionManager property)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDAFunctionManager.block">(warp_drive.managers.function_manager.CUDAFunctionManager property)</a>
</li>
      </ul></li>
  </ul></td>
</tr></table>

<h2 id="C">C</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="warp_drive.utils.html#warp_drive.utils.common.check_env_header">check_env_header() (in module warp_drive.utils.common)</a>
</li>
      <li><a href="function_manager.html#function_manager.CUDAFunctionManager.compile_and_load_cuda">compile_and_load_cuda() (function_manager.CUDAFunctionManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDAFunctionManager.compile_and_load_cuda">(warp_drive.managers.function_manager.CUDAFunctionManager method)</a>
</li>
      </ul></li>
      <li><a href="example_envs.tag_continuous.html#example_envs.tag_continuous.tag_continuous.TagContinuous.compute_distance">compute_distance() (example_envs.tag_continuous.tag_continuous.TagContinuous method)</a>
</li>
      <li><a href="warp_drive.training.algorithms.html#warp_drive.training.algorithms.a2c.A2C.compute_loss_and_metrics">compute_loss_and_metrics() (warp_drive.training.algorithms.a2c.A2C method)</a>

      <ul>
        <li><a href="warp_drive.training.algorithms.html#warp_drive.training.algorithms.ppo.PPO.compute_loss_and_metrics">(warp_drive.training.algorithms.ppo.PPO method)</a>
</li>
      </ul></li>
      <li><a href="example_envs.tag_continuous.html#example_envs.tag_continuous.tag_continuous.TagContinuous.compute_reward">compute_reward() (example_envs.tag_continuous.tag_continuous.TagContinuous method)</a>
</li>
      <li><a href="warp_drive.utils.html#warp_drive.utils.constants.Constants">Constants (class in warp_drive.utils.constants)</a>
</li>
      <li><a href="warp_drive.training.utils.html#warp_drive.training.utils.data_loader.create_and_push_data_placeholders">create_and_push_data_placeholders() (in module warp_drive.training.utils.data_loader)</a>
</li>
      <li><a href="function_manager.html#function_manager.CUDAFunctionManager.cuda_function_names">cuda_function_names (function_manager.CUDAFunctionManager property)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDAFunctionManager.cuda_function_names">(warp_drive.managers.function_manager.CUDAFunctionManager property)</a>
</li>
      </ul></li>
  </ul></td>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="data_manager.html#data_manager.CUDADataManager">CUDADataManager (class in data_manager)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CUDADataManager">(class in warp_drive.managers.data_manager)</a>
</li>
      </ul></li>
      <li><a href="function_manager.html#function_manager.CUDAEnvironmentReset">CUDAEnvironmentReset (class in function_manager)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDAEnvironmentReset">(class in warp_drive.managers.function_manager)</a>
</li>
      </ul></li>
      <li><a href="function_manager.html#function_manager.CUDAFunctionManager">CUDAFunctionManager (class in function_manager)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDAFunctionManager">(class in warp_drive.managers.function_manager)</a>
</li>
      </ul></li>
      <li><a href="function_manager.html#function_manager.CUDALogController">CUDALogController (class in function_manager)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDALogController">(class in warp_drive.managers.function_manager)</a>
</li>
      </ul></li>
      <li><a href="function_manager.html#function_manager.CUDASampler">CUDASampler (class in function_manager)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDASampler">(class in warp_drive.managers.function_manager)</a>
</li>
      </ul></li>
      <li><a href="data_manager.html#data_manager.CudaTensorHolder">CudaTensorHolder (class in data_manager)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CudaTensorHolder">(class in warp_drive.managers.data_manager)</a>
</li>
      </ul></li>
  </ul></td>
</tr></table>

<h2 id="D">D</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li>
    data_manager

      <ul>
        <li><a href="data_manager.html#module-data_manager">module</a>
</li>
      </ul></li>
      <li><a href="data_manager.html#data_manager.CUDADataManager.data_on_device_via_torch">data_on_device_via_torch() (data_manager.CUDADataManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CUDADataManager.data_on_device_via_torch">(warp_drive.managers.data_manager.CUDADataManager method)</a>
</li>
      </ul></li>
  </ul></td>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="warp_drive.utils.html#warp_drive.utils.data_feed.DataFeed">DataFeed (class in warp_drive.utils.data_feed)</a>
</li>
      <li><a href="data_manager.html#data_manager.CUDADataManager.device_data">device_data() (data_manager.CUDADataManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CUDADataManager.device_data">(warp_drive.managers.data_manager.CUDADataManager method)</a>
</li>
      </ul></li>
      <li><a href="warp_drive.utils.html#warp_drive.utils.constants.Constants.DONE_FLAGS">DONE_FLAGS (warp_drive.utils.constants.Constants attribute)</a>
</li>
  </ul></td>
</tr></table>

<h2 id="E">E</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="warp_drive.html#warp_drive.env_cpu_gpu_consistency_checker.EnvironmentCPUvsGPU">EnvironmentCPUvsGPU (class in warp_drive.env_cpu_gpu_consistency_checker)</a>
</li>
      <li><a href="warp_drive.html#warp_drive.env_wrapper.EnvWrapper">EnvWrapper (class in warp_drive.env_wrapper)</a>
</li>
      <li>
    example_envs

      <ul>
        <li><a href="example_envs.html#module-example_envs">module</a>
</li>
      </ul></li>
      <li>
    example_envs.tag_continuous

      <ul>
        <li><a href="example_envs.tag_continuous.html#module-example_envs.tag_continuous">module</a>
</li>
      </ul></li>
  </ul></td>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li>
    example_envs.tag_continuous.tag_continuous

      <ul>
        <li><a href="example_envs.tag_continuous.html#module-example_envs.tag_continuous.tag_continuous">module</a>
</li>
      </ul></li>
      <li>
    example_envs.tag_gridworld

      <ul>
        <li><a href="example_envs.tag_gridworld.html#module-example_envs.tag_gridworld">module</a>
</li>
      </ul></li>
      <li>
    example_envs.tag_gridworld.tag_gridworld

      <ul>
        <li><a href="example_envs.tag_gridworld.html#module-example_envs.tag_gridworld.tag_gridworld">module</a>
</li>
      </ul></li>
  </ul></td>
</tr></table>

<h2 id="F">F</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="warp_drive.training.html#warp_drive.training.trainer.Trainer.fetch_episode_global_states">fetch_episode_global_states() (warp_drive.training.trainer.Trainer method)</a>
</li>
      <li><a href="function_manager.html#function_manager.CUDALogController.fetch_log">fetch_log() (function_manager.CUDALogController method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDALogController.fetch_log">(warp_drive.managers.function_manager.CUDALogController method)</a>
</li>
      </ul></li>
  </ul></td>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="warp_drive.training.models.html#warp_drive.training.models.fully_connected.FullyConnected.forward">forward() (warp_drive.training.models.fully_connected.FullyConnected method)</a>
</li>
      <li><a href="warp_drive.training.models.html#warp_drive.training.models.fully_connected.FullyConnected">FullyConnected (class in warp_drive.training.models.fully_connected)</a>
</li>
      <li>
    function_manager

      <ul>
        <li><a href="function_manager.html#module-function_manager">module</a>
</li>
      </ul></li>
  </ul></td>
</tr></table>

<h2 id="G">G</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="example_envs.tag_continuous.html#example_envs.tag_continuous.tag_continuous.TagContinuous.generate_observation">generate_observation() (example_envs.tag_continuous.tag_continuous.TagContinuous method)</a>

      <ul>
        <li><a href="example_envs.tag_gridworld.html#example_envs.tag_gridworld.tag_gridworld.TagGridWorld.generate_observation">(example_envs.tag_gridworld.tag_gridworld.TagGridWorld method)</a>
</li>
      </ul></li>
      <li><a href="warp_drive.html#warp_drive.env_cpu_gpu_consistency_checker.generate_random_actions">generate_random_actions() (in module warp_drive.env_cpu_gpu_consistency_checker)</a>
</li>
      <li><a href="warp_drive.training.html#warp_drive.training.trainer.Trainer.generate_rollout">generate_rollout() (warp_drive.training.trainer.Trainer method)</a>
</li>
      <li><a href="warp_drive.training.html#warp_drive.training.trainer.PerfStats.get">get() (warp_drive.training.trainer.PerfStats method)</a>
</li>
      <li><a href="example_envs.tag_continuous.html#example_envs.tag_continuous.tag_continuous.TagContinuous.get_data_dictionary">get_data_dictionary() (example_envs.tag_continuous.tag_continuous.TagContinuous method)</a>

      <ul>
        <li><a href="example_envs.tag_gridworld.html#example_envs.tag_gridworld.tag_gridworld.TagGridWorld.get_data_dictionary">(example_envs.tag_gridworld.tag_gridworld.TagGridWorld method)</a>
</li>
      </ul></li>
      <li><a href="data_manager.html#data_manager.CUDADataManager.get_dtype">get_dtype() (data_manager.CUDADataManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CUDADataManager.get_dtype">(warp_drive.managers.data_manager.CUDADataManager method)</a>
</li>
      </ul></li>
  </ul></td>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="warp_drive.utils.html#warp_drive.utils.common.get_env_directory">get_env_directory() (in module warp_drive.utils.common)</a>
</li>
      <li><a href="warp_drive.utils.html#warp_drive.utils.common.get_project_root">get_project_root() (in module warp_drive.utils.common)</a>
</li>
      <li><a href="data_manager.html#data_manager.CUDADataManager.get_shape">get_shape() (data_manager.CUDADataManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CUDADataManager.get_shape">(warp_drive.managers.data_manager.CUDADataManager method)</a>
</li>
      </ul></li>
      <li><a href="example_envs.tag_continuous.html#example_envs.tag_continuous.tag_continuous.TagContinuous.get_tensor_dictionary">get_tensor_dictionary() (example_envs.tag_continuous.tag_continuous.TagContinuous method)</a>

      <ul>
        <li><a href="example_envs.tag_gridworld.html#example_envs.tag_gridworld.tag_gridworld.TagGridWorld.get_tensor_dictionary">(example_envs.tag_gridworld.tag_gridworld.TagGridWorld method)</a>
</li>
      </ul></li>
      <li><a href="warp_drive.training.html#warp_drive.training.trainer.Trainer.graceful_close">graceful_close() (warp_drive.training.trainer.Trainer method)</a>
</li>
      <li><a href="function_manager.html#function_manager.CUDAFunctionManager.grid">grid (function_manager.CUDAFunctionManager property)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDAFunctionManager.grid">(warp_drive.managers.function_manager.CUDAFunctionManager property)</a>
</li>
      </ul></li>
  </ul></td>
</tr></table>

<h2 id="H">H</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="data_manager.html#data_manager.CUDADataManager.host_data">host_data (data_manager.CUDADataManager property)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CUDADataManager.host_data">(warp_drive.managers.data_manager.CUDADataManager property)</a>
</li>
      </ul></li>
  </ul></td>
</tr></table>

<h2 id="I">I</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="function_manager.html#function_manager.CUDASampler.init_random">init_random() (function_manager.CUDASampler method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDASampler.init_random">(warp_drive.managers.function_manager.CUDASampler method)</a>
</li>
      </ul></li>
      <li><a href="function_manager.html#function_manager.CUDAFunctionManager.initialize_default_functions">initialize_default_functions() (function_manager.CUDAFunctionManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDAFunctionManager.initialize_default_functions">(warp_drive.managers.function_manager.CUDAFunctionManager method)</a>
</li>
      </ul></li>
      <li><a href="function_manager.html#function_manager.CUDAFunctionManager.initialize_functions">initialize_functions() (function_manager.CUDAFunctionManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDAFunctionManager.initialize_functions">(warp_drive.managers.function_manager.CUDAFunctionManager method)</a>
</li>
      </ul></li>
  </ul></td>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="function_manager.html#function_manager.CUDAFunctionManager.initialize_shared_constants">initialize_shared_constants() (function_manager.CUDAFunctionManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDAFunctionManager.initialize_shared_constants">(warp_drive.managers.function_manager.CUDAFunctionManager method)</a>
</li>
      </ul></li>
      <li><a href="data_manager.html#data_manager.CUDADataManager.is_data_on_device">is_data_on_device() (data_manager.CUDADataManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CUDADataManager.is_data_on_device">(warp_drive.managers.data_manager.CUDADataManager method)</a>
</li>
      </ul></li>
      <li><a href="data_manager.html#data_manager.CUDADataManager.is_data_on_device_via_torch">is_data_on_device_via_torch() (data_manager.CUDADataManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CUDADataManager.is_data_on_device_via_torch">(warp_drive.managers.data_manager.CUDADataManager method)</a>
</li>
      </ul></li>
  </ul></td>
</tr></table>

<h2 id="K">K</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="example_envs.tag_continuous.html#example_envs.tag_continuous.tag_continuous.TagContinuous.K_nearest_neighbors">K_nearest_neighbors() (example_envs.tag_continuous.tag_continuous.TagContinuous method)</a>
</li>
  </ul></td>
</tr></table>

<h2 id="L">L</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="function_manager.html#function_manager.CUDAFunctionManager.load_cuda_from_binary_file">load_cuda_from_binary_file() (function_manager.CUDAFunctionManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDAFunctionManager.load_cuda_from_binary_file">(warp_drive.managers.function_manager.CUDAFunctionManager method)</a>
</li>
      </ul></li>
      <li><a href="function_manager.html#function_manager.CUDAFunctionManager.load_cuda_from_source_code">load_cuda_from_source_code() (function_manager.CUDAFunctionManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDAFunctionManager.load_cuda_from_source_code">(warp_drive.managers.function_manager.CUDAFunctionManager method)</a>
</li>
      </ul></li>
  </ul></td>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="warp_drive.training.html#warp_drive.training.trainer.Trainer.load_model_checkpoint">load_model_checkpoint() (warp_drive.training.trainer.Trainer method)</a>
</li>
      <li><a href="data_manager.html#data_manager.CUDADataManager.log_data_list">log_data_list (data_manager.CUDADataManager property)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CUDADataManager.log_data_list">(warp_drive.managers.data_manager.CUDADataManager property)</a>
</li>
      </ul></li>
  </ul></td>
</tr></table>

<h2 id="M">M</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="data_manager.html#data_manager.CUDADataManager.meta_info">meta_info() (data_manager.CUDADataManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CUDADataManager.meta_info">(warp_drive.managers.data_manager.CUDADataManager method)</a>
</li>
      </ul></li>
      <li><a href="warp_drive.training.html#warp_drive.training.trainer.Metrics">Metrics (class in warp_drive.training.trainer)</a>
</li>
      <li>
    module

      <ul>
        <li><a href="data_manager.html#module-data_manager">data_manager</a>
</li>
        <li><a href="example_envs.html#module-example_envs">example_envs</a>
</li>
        <li><a href="example_envs.tag_continuous.html#module-example_envs.tag_continuous">example_envs.tag_continuous</a>
</li>
        <li><a href="example_envs.tag_continuous.html#module-example_envs.tag_continuous.tag_continuous">example_envs.tag_continuous.tag_continuous</a>
</li>
        <li><a href="example_envs.tag_gridworld.html#module-example_envs.tag_gridworld">example_envs.tag_gridworld</a>
</li>
        <li><a href="example_envs.tag_gridworld.html#module-example_envs.tag_gridworld.tag_gridworld">example_envs.tag_gridworld.tag_gridworld</a>
</li>
        <li><a href="function_manager.html#module-function_manager">function_manager</a>
</li>
        <li><a href="warp_drive.html#module-warp_drive">warp_drive</a>
</li>
        <li><a href="warp_drive.html#module-warp_drive.env_cpu_gpu_consistency_checker">warp_drive.env_cpu_gpu_consistency_checker</a>
</li>
        <li><a href="warp_drive.html#module-warp_drive.env_wrapper">warp_drive.env_wrapper</a>
</li>
        <li><a href="warp_drive.managers.html#module-warp_drive.managers">warp_drive.managers</a>
</li>
        <li><a href="warp_drive.managers.html#module-warp_drive.managers.data_manager">warp_drive.managers.data_manager</a>
</li>
        <li><a href="warp_drive.managers.html#module-warp_drive.managers.function_manager">warp_drive.managers.function_manager</a>
</li>
        <li><a href="warp_drive.training.html#module-warp_drive.training">warp_drive.training</a>
</li>
        <li><a href="warp_drive.training.algorithms.html#module-warp_drive.training.algorithms">warp_drive.training.algorithms</a>
</li>
        <li><a href="warp_drive.training.algorithms.html#module-warp_drive.training.algorithms.a2c">warp_drive.training.algorithms.a2c</a>
</li>
        <li><a href="warp_drive.training.algorithms.html#module-warp_drive.training.algorithms.ppo">warp_drive.training.algorithms.ppo</a>
</li>
        <li><a href="warp_drive.training.models.html#module-warp_drive.training.models">warp_drive.training.models</a>
</li>
        <li><a href="warp_drive.training.models.html#module-warp_drive.training.models.fully_connected">warp_drive.training.models.fully_connected</a>
</li>
        <li><a href="warp_drive.training.html#module-warp_drive.training.trainer">warp_drive.training.trainer</a>
</li>
        <li><a href="warp_drive.training.utils.html#module-warp_drive.training.utils">warp_drive.training.utils</a>
</li>
        <li><a href="warp_drive.training.utils.html#module-warp_drive.training.utils.data_loader">warp_drive.training.utils.data_loader</a>
</li>
        <li><a href="warp_drive.utils.html#module-warp_drive.utils">warp_drive.utils</a>
</li>
        <li><a href="warp_drive.utils.html#module-warp_drive.utils.common">warp_drive.utils.common</a>
</li>
        <li><a href="warp_drive.utils.html#module-warp_drive.utils.constants">warp_drive.utils.constants</a>
</li>
        <li><a href="warp_drive.utils.html#module-warp_drive.utils.data_feed">warp_drive.utils.data_feed</a>
</li>
        <li><a href="warp_drive.utils.html#module-warp_drive.utils.recursive_obs_dict_to_spaces_dict">warp_drive.utils.recursive_obs_dict_to_spaces_dict</a>
</li>
      </ul></li>
  </ul></td>
</tr></table>

<h2 id="N">N</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="example_envs.tag_continuous.html#example_envs.tag_continuous.tag_continuous.TagContinuous.name">name (example_envs.tag_continuous.tag_continuous.TagContinuous attribute)</a>

      <ul>
        <li><a href="example_envs.tag_gridworld.html#example_envs.tag_gridworld.tag_gridworld.TagGridWorld.name">(example_envs.tag_gridworld.tag_gridworld.TagGridWorld attribute)</a>
</li>
        <li><a href="warp_drive.training.models.html#warp_drive.training.models.fully_connected.FullyConnected.name">(warp_drive.training.models.fully_connected.FullyConnected attribute)</a>
</li>
      </ul></li>
  </ul></td>
</tr></table>

<h2 id="O">O</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="warp_drive.utils.html#warp_drive.utils.constants.Constants.OBSERVATIONS">OBSERVATIONS (warp_drive.utils.constants.Constants attribute)</a>
</li>
  </ul></td>
</tr></table>

<h2 id="P">P</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="warp_drive.training.html#warp_drive.training.trainer.PerfStats">PerfStats (class in warp_drive.training.trainer)</a>
</li>
      <li><a href="warp_drive.training.algorithms.html#warp_drive.training.algorithms.ppo.PPO">PPO (class in warp_drive.training.algorithms.ppo)</a>
</li>
      <li><a href="warp_drive.training.html#warp_drive.training.trainer.Metrics.pretty_print">pretty_print() (warp_drive.training.trainer.Metrics method)</a>

      <ul>
        <li><a href="warp_drive.training.html#warp_drive.training.trainer.PerfStats.pretty_print">(warp_drive.training.trainer.PerfStats method)</a>
</li>
      </ul></li>
  </ul></td>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="data_manager.html#data_manager.CUDADataManager.pull_data_from_device">pull_data_from_device() (data_manager.CUDADataManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CUDADataManager.pull_data_from_device">(warp_drive.managers.data_manager.CUDADataManager method)</a>
</li>
      </ul></li>
      <li><a href="data_manager.html#data_manager.CUDADataManager.push_data_to_device">push_data_to_device() (data_manager.CUDADataManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CUDADataManager.push_data_to_device">(warp_drive.managers.data_manager.CUDADataManager method)</a>
</li>
      </ul></li>
  </ul></td>
</tr></table>

<h2 id="R">R</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="warp_drive.utils.html#warp_drive.utils.recursive_obs_dict_to_spaces_dict.recursive_obs_dict_to_spaces_dict">recursive_obs_dict_to_spaces_dict() (in module warp_drive.utils.recursive_obs_dict_to_spaces_dict)</a>
</li>
      <li><a href="function_manager.html#function_manager.CUDASampler.register_actions">register_actions() (function_manager.CUDASampler method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDASampler.register_actions">(warp_drive.managers.function_manager.CUDASampler method)</a>
</li>
      </ul></li>
      <li><a href="example_envs.tag_continuous.html#example_envs.tag_continuous.tag_continuous.TagContinuous.reset">reset() (example_envs.tag_continuous.tag_continuous.TagContinuous method)</a>

      <ul>
        <li><a href="example_envs.tag_gridworld.html#example_envs.tag_gridworld.tag_gridworld.TagGridWorld.reset">(example_envs.tag_gridworld.tag_gridworld.TagGridWorld method)</a>
</li>
      </ul></li>
      <li><a href="warp_drive.html#warp_drive.env_wrapper.EnvWrapper.reset_all_envs">reset_all_envs() (warp_drive.env_wrapper.EnvWrapper method)</a>
</li>
      <li><a href="data_manager.html#data_manager.CUDADataManager.reset_data_list">reset_data_list (data_manager.CUDADataManager property)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CUDADataManager.reset_data_list">(warp_drive.managers.data_manager.CUDADataManager property)</a>
</li>
      </ul></li>
  </ul></td>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="data_manager.html#data_manager.CUDADataManager.reset_device">reset_device() (data_manager.CUDADataManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CUDADataManager.reset_device">(warp_drive.managers.data_manager.CUDADataManager method)</a>
</li>
      </ul></li>
      <li><a href="function_manager.html#function_manager.CUDALogController.reset_log">reset_log() (function_manager.CUDALogController method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDALogController.reset_log">(warp_drive.managers.function_manager.CUDALogController method)</a>
</li>
      </ul></li>
      <li><a href="warp_drive.html#warp_drive.env_wrapper.EnvWrapper.reset_only_done_envs">reset_only_done_envs() (warp_drive.env_wrapper.EnvWrapper method)</a>
</li>
      <li><a href="function_manager.html#function_manager.CUDAEnvironmentReset.reset_when_done">reset_when_done() (function_manager.CUDAEnvironmentReset method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDAEnvironmentReset.reset_when_done">(warp_drive.managers.function_manager.CUDAEnvironmentReset method)</a>
</li>
      </ul></li>
      <li><a href="warp_drive.utils.html#warp_drive.utils.constants.Constants.REWARDS">REWARDS (warp_drive.utils.constants.Constants attribute)</a>
</li>
      <li><a href="warp_drive.html#warp_drive.env_cpu_gpu_consistency_checker.EnvironmentCPUvsGPU.run_consistency_checks">run_consistency_checks() (warp_drive.env_cpu_gpu_consistency_checker.EnvironmentCPUvsGPU static method)</a>
</li>
  </ul></td>
</tr></table>

<h2 id="S">S</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="function_manager.html#function_manager.CUDASampler.sample">sample() (function_manager.CUDASampler method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDASampler.sample">(warp_drive.managers.function_manager.CUDASampler method)</a>
</li>
      </ul></li>
      <li><a href="warp_drive.training.html#warp_drive.training.trainer.Trainer.save_model_checkpoint">save_model_checkpoint() (warp_drive.training.trainer.Trainer method)</a>
</li>
      <li><a href="data_manager.html#data_manager.CUDADataManager.scalar_data_list">scalar_data_list (data_manager.CUDADataManager property)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CUDADataManager.scalar_data_list">(warp_drive.managers.data_manager.CUDADataManager property)</a>
</li>
      </ul></li>
      <li><a href="example_envs.tag_continuous.html#example_envs.tag_continuous.tag_continuous.TagContinuous.seed">seed() (example_envs.tag_continuous.tag_continuous.TagContinuous method)</a>

      <ul>
        <li><a href="example_envs.tag_gridworld.html#example_envs.tag_gridworld.tag_gridworld.TagGridWorld.seed">(example_envs.tag_gridworld.tag_gridworld.TagGridWorld method)</a>
</li>
      </ul></li>
  </ul></td>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="warp_drive.training.models.html#warp_drive.training.models.fully_connected.FullyConnected.set_fast_forward_mode">set_fast_forward_mode() (warp_drive.training.models.fully_connected.FullyConnected method)</a>
</li>
      <li><a href="example_envs.tag_continuous.html#example_envs.tag_continuous.tag_continuous.TagContinuous.set_global_state">set_global_state() (example_envs.tag_continuous.tag_continuous.TagContinuous method)</a>

      <ul>
        <li><a href="example_envs.tag_gridworld.html#example_envs.tag_gridworld.tag_gridworld.TagGridWorld.set_global_state">(example_envs.tag_gridworld.tag_gridworld.TagGridWorld method)</a>
</li>
      </ul></li>
      <li><a href="data_manager.html#data_manager.CUDADataManager.shared_constant">shared_constant() (data_manager.CUDADataManager method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.data_manager.CUDADataManager.shared_constant">(warp_drive.managers.data_manager.CUDADataManager method)</a>
</li>
      </ul></li>
      <li><a href="example_envs.tag_continuous.html#example_envs.tag_continuous.tag_continuous.TagContinuous.step">step() (example_envs.tag_continuous.tag_continuous.TagContinuous method)</a>

      <ul>
        <li><a href="example_envs.tag_gridworld.html#example_envs.tag_gridworld.tag_gridworld.TagGridWorld.step">(example_envs.tag_gridworld.tag_gridworld.TagGridWorld method)</a>
</li>
        <li><a href="warp_drive.html#warp_drive.env_wrapper.EnvWrapper.step">(warp_drive.env_wrapper.EnvWrapper method)</a>
</li>
      </ul></li>
  </ul></td>
</tr></table>

<h2 id="T">T</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="example_envs.tag_continuous.html#example_envs.tag_continuous.tag_continuous.TagContinuous">TagContinuous (class in example_envs.tag_continuous.tag_continuous)</a>
</li>
      <li><a href="example_envs.tag_gridworld.html#example_envs.tag_gridworld.tag_gridworld.TagGridWorld">TagGridWorld (class in example_envs.tag_gridworld.tag_gridworld)</a>
</li>
      <li><a href="warp_drive.html#warp_drive.env_cpu_gpu_consistency_checker.EnvironmentCPUvsGPU.test_env_reset_and_step">test_env_reset_and_step() (warp_drive.env_cpu_gpu_consistency_checker.EnvironmentCPUvsGPU method)</a>
</li>
  </ul></td>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="warp_drive.training.html#warp_drive.training.trainer.Trainer.train">train() (warp_drive.training.trainer.Trainer method)</a>
</li>
      <li><a href="warp_drive.training.html#warp_drive.training.trainer.Trainer">Trainer (class in warp_drive.training.trainer)</a>
</li>
      <li><a href="warp_drive.training.models.html#warp_drive.training.models.fully_connected.FullyConnected.training">training (warp_drive.training.models.fully_connected.FullyConnected attribute)</a>
</li>
  </ul></td>
</tr></table>

<h2 id="U">U</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="warp_drive.utils.html#warp_drive.utils.common.update_env_header">update_env_header() (in module warp_drive.utils.common)</a>
</li>
      <li><a href="warp_drive.utils.html#warp_drive.utils.common.update_env_runner">update_env_runner() (in module warp_drive.utils.common)</a>
</li>
      <li><a href="function_manager.html#function_manager.CUDALogController.update_log">update_log() (function_manager.CUDALogController method)</a>

      <ul>
        <li><a href="warp_drive.managers.html#warp_drive.managers.function_manager.CUDALogController.update_log">(warp_drive.managers.function_manager.CUDALogController method)</a>
</li>
      </ul></li>
  </ul></td>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="example_envs.tag_continuous.html#example_envs.tag_continuous.tag_continuous.TagContinuous.update_state">update_state() (example_envs.tag_continuous.tag_continuous.TagContinuous method)</a>

      <ul>
        <li><a href="example_envs.tag_gridworld.html#example_envs.tag_gridworld.tag_gridworld.TagGridWorld.update_state">(example_envs.tag_gridworld.tag_gridworld.TagGridWorld method)</a>
</li>
      </ul></li>
  </ul></td>
</tr></table>

<h2 id="W">W</h2>
<table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li>
    warp_drive

      <ul>
        <li><a href="warp_drive.html#module-warp_drive">module</a>
</li>
      </ul></li>
      <li>
    warp_drive.env_cpu_gpu_consistency_checker

      <ul>
        <li><a href="warp_drive.html#module-warp_drive.env_cpu_gpu_consistency_checker">module</a>
</li>
      </ul></li>
      <li>
    warp_drive.env_wrapper

      <ul>
        <li><a href="warp_drive.html#module-warp_drive.env_wrapper">module</a>
</li>
      </ul></li>
      <li>
    warp_drive.managers

      <ul>
        <li><a href="warp_drive.managers.html#module-warp_drive.managers">module</a>
</li>
      </ul></li>
      <li>
    warp_drive.managers.data_manager

      <ul>
        <li><a href="warp_drive.managers.html#module-warp_drive.managers.data_manager">module</a>
</li>
      </ul></li>
      <li>
    warp_drive.managers.function_manager

      <ul>
        <li><a href="warp_drive.managers.html#module-warp_drive.managers.function_manager">module</a>
</li>
      </ul></li>
      <li>
    warp_drive.training

      <ul>
        <li><a href="warp_drive.training.html#module-warp_drive.training">module</a>
</li>
      </ul></li>
      <li>
    warp_drive.training.algorithms

      <ul>
        <li><a href="warp_drive.training.algorithms.html#module-warp_drive.training.algorithms">module</a>
</li>
      </ul></li>
      <li>
    warp_drive.training.algorithms.a2c

      <ul>
        <li><a href="warp_drive.training.algorithms.html#module-warp_drive.training.algorithms.a2c">module</a>
</li>
      </ul></li>
      <li>
    warp_drive.training.algorithms.ppo

      <ul>
        <li><a href="warp_drive.training.algorithms.html#module-warp_drive.training.algorithms.ppo">module</a>
</li>
      </ul></li>
  </ul></td>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li>
    warp_drive.training.models

      <ul>
        <li><a href="warp_drive.training.models.html#module-warp_drive.training.models">module</a>
</li>
      </ul></li>
      <li>
    warp_drive.training.models.fully_connected

      <ul>
        <li><a href="warp_drive.training.models.html#module-warp_drive.training.models.fully_connected">module</a>
</li>
      </ul></li>
      <li>
    warp_drive.training.trainer

      <ul>
        <li><a href="warp_drive.training.html#module-warp_drive.training.trainer">module</a>
</li>
      </ul></li>
      <li>
    warp_drive.training.utils

      <ul>
        <li><a href="warp_drive.training.utils.html#module-warp_drive.training.utils">module</a>
</li>
      </ul></li>
      <li>
    warp_drive.training.utils.data_loader

      <ul>
        <li><a href="warp_drive.training.utils.html#module-warp_drive.training.utils.data_loader">module</a>
</li>
      </ul></li>
      <li>
    warp_drive.utils

      <ul>
        <li><a href="warp_drive.utils.html#module-warp_drive.utils">module</a>
</li>
      </ul></li>
      <li>
    warp_drive.utils.common

      <ul>
        <li><a href="warp_drive.utils.html#module-warp_drive.utils.common">module</a>
</li>
      </ul></li>
      <li>
    warp_drive.utils.constants

      <ul>
        <li><a href="warp_drive.utils.html#module-warp_drive.utils.constants">module</a>
</li>
      </ul></li>
      <li>
    warp_drive.utils.data_feed

      <ul>
        <li><a href="warp_drive.utils.html#module-warp_drive.utils.data_feed">module</a>
</li>
      </ul></li>
      <li>
    warp_drive.utils.recursive_obs_dict_to_spaces_dict

      <ul>
        <li><a href="warp_drive.utils.html#module-warp_drive.utils.recursive_obs_dict_to_spaces_dict">module</a>
</li>
      </ul></li>
  </ul></td>
</tr></table>



           </div>
           
          </div>
          <footer>

  <hr/>

  <div role="contentinfo">
    <p>
        &#169; Copyright 2021, Tian Lan, Sunil Srinivasa, Stephan Zheng.

    </p>
  </div>
    
    
    
    Built with <a href="https://www.sphinx-doc.org/">Sphinx</a> using a
    
    <a href="https://github.com/readthedocs/sphinx_rtd_theme">theme</a>
    
    provided by <a href="https://readthedocs.org">Read the Docs</a>. 

</footer>
        </div>
      </div>

    </section>

  </div>
  

  <script type="text/javascript">
      jQuery(function () {
          SphinxRtdTheme.Navigation.enable(true);
      });
  </script>

  
  
    
   

</body>
</html>